/*************************************************************************
> File Name: vectorAndHashMap.cpp
> Author: icebear
> Mail: 3322366195@qq.com
> Created Time: 2024-01-14 14:24:16
> Description:
380. O(1) 时间插入、删除和获取随机元素
实现RandomizedSet 类：
RandomizedSet() 初始化 RandomizedSet 对象
bool insert(int val) 当元素 val 不存在时，向集合中插入该项，并返回 true ；否则，返回 false 。
bool remove(int val) 当元素 val 存在时，从集合中移除该项，并返回 true ；否则，返回 false 。
int getRandom() 随机返回现有集合中的一项（测试用例保证调用此方法时集合中至少存在一个元素）。每个元素应该有 相同的概率 被返回。
你必须实现类的所有函数，并满足每个函数的 平均 时间复杂度为 O(1) 。
************************************************************************/

// 思路:
// vector 可以实现 随即元素访问O(1)
// hashmap 可以实现 插入和 删除O(1)
// 综合

#include <vector>
#include <unordered_map>
#include <stdlib.h>
#include <time.h>
using namespace std;

class RandomizedSet
{
private:
    vector<int> nums;
    unordered_map<int, int> map;
    int size;

public:
    RandomizedSet()
    {
        // srand((unsigned)time(NULL));
        size = 0;
    }

    bool insert(int val)
    {
        if (map.find(val) != map.end())
        {
            return false;
        }
        nums.push_back(val);
        map[val] = size;
        size++;
        return true;
    }

    bool remove(int val)
    {
        if (map.find(val) == map.end())
        {
            return false;
        }
        int index = map[val];
        nums[index] = nums[size-1];
        map[nums[size-1]] = index;
        map.erase(val);
        nums.pop_back();
        size--;
        return true;
    }

    int getRandom()
    {
        int index = rand() % nums.size();
        return nums[index];
    }
};

/**
 * Your RandomizedSet object will be instantiated and called as such:
 * RandomizedSet* obj = new RandomizedSet();
 * bool param_1 = obj->insert(val);
 * bool param_2 = obj->remove(val);
 * int param_3 = obj->getRandom();
 */